package com.sevenGroup.pawnshop.Utils;

import com.baomidou.mybatisplus.core.toolkit.StringPool;
import com.baomidou.mybatisplus.generator.AutoGenerator;
import com.baomidou.mybatisplus.generator.InjectionConfig;
import com.baomidou.mybatisplus.generator.config.*;
import com.baomidou.mybatisplus.generator.config.po.TableInfo;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
import com.baomidou.mybatisplus.generator.engine.VelocityTemplateEngine;

import java.util.ArrayList;
import java.util.List;

public class Auto {

        public static void main(String[] args) {
            // 代码生成器
            AutoGenerator mpg = new AutoGenerator();

            // 全局配置
            GlobalConfig gc = new GlobalConfig();
            String projectPath = System.getProperty("user.dir");//获取项目的根路径
            gc.setOutputDir(projectPath + "/pawnshop/src/main/java");//注意要加模块名称
            gc.setAuthor("sevenGroup");//设置作者
            gc.setMapperName("%sDao");
            gc.setServiceName("%sService");//取消生成的业务层类名前的I前缀
            gc.setOpen(false);//设置自动生成代码后的包是否自动打开
            //gc.setFileOverride(false);//设置是否覆盖之前的代码
            // gc.setSwagger2(true); 实体属性 Swagger2 注解
            mpg.setGlobalConfig(gc);

            // 数据源配置
            DataSourceConfig dsc = new DataSourceConfig();
            //设置代码生成时，找的数据库是谁
            dsc.setUrl("jdbc:mysql://124.222.44.178:3306/pawnshops?useUnicode=true&characterEncoding=utf8&serverTimezone=GMT%2B8&useSSL=false");
            // dsc.setSchemaName("public");
            //设置驱动
            dsc.setDriverName("com.mysql.cj.jdbc.Driver");
            //设置用户名
            dsc.setUsername("root");
            //设置密码
            dsc.setPassword("zh640832");
            mpg.setDataSource(dsc);

            // 包配置
            PackageConfig pc = new PackageConfig();
            pc.setEntity("model");//设置生成的实体类的包名
            pc.setMapper("dao");//设置生成数据访问层的包名
            pc.setService("service");//设置生成的业务层的包名
            pc.setController("controller");//设置生成控制层的包名
            pc.setParent("com.sevenGroup.pawnshop");//设置生成的代码放到哪个包下
            mpg.setPackageInfo(pc);

            // 自定义配置
            InjectionConfig cfg = new InjectionConfig() {
                @Override
                public void initMap() {
                    // to do nothing
                }
            };

            // 如果模板引擎是 freemarker
            //String templatePath = "/templates/mapper.xml.ftl";
            // 如果模板引擎是 velocity
            String templatePath = "/templates/mapper.xml.vm";

            // 自定义输出配置
            List<FileOutConfig> focList = new ArrayList<>();
            // 自定义配置会被优先输出
            focList.add(new FileOutConfig(templatePath) {
                @Override
                public String outputFile(TableInfo tableInfo) {
                    // 设置生成的xml映射文件放置的位置及名称
                    return projectPath + "/sevenGroup/src/main/resources/mappers/"
                            + "/" + tableInfo.getEntityName() + "Dao" + StringPool.DOT_XML;
                }
            });
            cfg.setFileOutConfigList(focList);
            mpg.setCfg(cfg);

            // 配置模板
            TemplateConfig templateConfig = new TemplateConfig();

            // 配置自定义输出模板
            //指定自定义模板路径，注意不要带上.ftl/.vm, 会根据使用的模板引擎自动识别
            // templateConfig.setEntity("templates/entity2.java");
            // templateConfig.setService();
            // templateConfig.setController();

            templateConfig.setXml(null);//设置生成的xml映射文件内容为空
            mpg.setTemplate(templateConfig);

            // 策略配置
            StrategyConfig strategy = new StrategyConfig();
            strategy.setInclude("goods_properties","first_type_properties","properties");//指定生成哪些表的内容
            strategy.setNaming(NamingStrategy.underline_to_camel);//设置下划线转驼峰
            strategy.setColumnNaming(NamingStrategy.underline_to_camel);//设置下划线转驼峰

            strategy.setEntityLombokModel(true);//设置实体类上使用lombok注解
            strategy.setRestControllerStyle(true);//设置控制器使用@RestController

//        strategy.setTablePrefix("menu_","account_");//去除生成的类的表的前缀
            mpg.setStrategy(strategy);
            mpg.setTemplateEngine(new VelocityTemplateEngine());
            mpg.execute();
        }
}
